package com.cl.cloud.gateway.filter;

import com.cl.cloud.gateway.util.RequestUtil;
import com.cl.core.common.ClCloudYamlPropertiesContant;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.boot.actuate.autoconfigure.endpoint.web.WebEndpointProperties;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.http.HttpHeaders;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

import java.util.ArrayList;
import java.util.List;

/**
 * @description: 用来打印请求信息的log
 * @author: liuzijian
 * @time: 2022-04-15 10:28
 */
@Slf4j
@Component
@AllArgsConstructor
@Order(Ordered.LOWEST_PRECEDENCE)
@ConditionalOnProperty(value = ClCloudYamlPropertiesContant.CL_CLOUD_LOG_ENABLED, havingValue = "true", matchIfMissing = true)
public class RequestLogFilter implements GlobalFilter {

    private final WebEndpointProperties endpointProperties;

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
        // 打印请求路径
        String path = request.getPath().pathWithinApplication().value();

        // 忽略 endpoint 请求
        String endpointBasePath = endpointProperties.getBasePath();
        if (StringUtils.isNotBlank(endpointBasePath) && path.startsWith(endpointBasePath)) {
            return chain.filter(exchange);
        }

        String requestUrl = RequestUtil.getOriginalRequestUrl(exchange);

        // 构建成一条长 日志，避免并发下日志错乱
        StringBuilder beforeReqLog = new StringBuilder(300);
        // 日志参数
        List<Object> beforeReqArgs = new ArrayList<>();
        beforeReqLog.append("\n\n================ Gateway Request Start  ================\n");
        // 打印路由
        beforeReqLog.append("===> {}: {}\n");
        // 参数
        String requestMethod = request.getMethodValue();
        beforeReqArgs.add(requestMethod);
        beforeReqArgs.add(requestUrl);

        // 打印请求头
        HttpHeaders headers = request.getHeaders();
        headers.forEach((headerName, headerValue) -> {
            beforeReqLog.append("===Headers===  {}: {}\n");
            beforeReqArgs.add(headerName);
//            if (AuthProvider.AUTH_KEY.toLowerCase().equals(headerName)) {
//                String value = headerValue.get(0);
//                String token = JwtUtil.getToken(value);
//                Claims claims = JwtUtil.parseJWT(token);
//                beforeReqArgs.add((claims == null) ? "" : claims.toString());
//                beforeReqLog.append("===Headers===  {}: {}\n");
//                beforeReqArgs.add(headerName.concat("-original"));
//                beforeReqArgs.add(StringUtils.join(headerValue.toArray()));
//            } else {
//                beforeReqArgs.add(StringUtils.join(headerValue.toArray()));
//            }
            beforeReqArgs.add(StringUtils.join(headerValue.toArray()));
        });

        beforeReqLog.append("================  Gateway Request End  =================\n");
        // 打印执行时间
        log.info(beforeReqLog.toString(), beforeReqArgs.toArray());
        return chain.filter(exchange);
    }
}
